package leetcode.dynamic;

import java.util.Arrays;

/**
 * 在一个火车旅行很受欢迎的国度，你提前一年计划了一些火车旅行。在接下来的一年里，你要旅行的日子将以一个名为 days 的数组给出。每一项是一个从 1 到 365 的整数。
 * <p>
 * 火车票有 三种不同的销售方式 ：
 * <p>
 * 一张 为期一天 的通行证售价为 costs[0] 美元；
 * 一张 为期七天 的通行证售价为 costs[1] 美元；
 * 一张 为期三十天 的通行证售价为 costs[2] 美元。
 * 通行证允许数天无限制的旅行。 例如，如果我们在第 2 天获得一张 为期 7 天 的通行证，那么我们可以连着旅行 7 天：第 2 天、第 3 天、第 4 天、第 5 天、第 6 天、第 7 天和第 8 天。
 * <p>
 * 返回 你想要完成在给定的列表 days 中列出的每一天的旅行所需要的最低消费 。
 * <p>
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * 输入：days = [1,4,6,7,8,20], costs = [2,7,15]
 * 输出：11
 * 解释：
 * 例如，这里有一种购买通行证的方法，可以让你完成你的旅行计划：
 * 在第 1 天，你花了 costs[0] = $2 买了一张为期 1 天的通行证，它将在第 1 天生效。
 * 在第 3 天，你花了 costs[1] = $7 买了一张为期 7 天的通行证，它将在第 3, 4, ..., 9 天生效。
 * 在第 20 天，你花了 costs[0] = $2 买了一张为期 1 天的通行证，它将在第 20 天生效。
 * 你总共花了 $11，并完成了你计划的每一天旅行。
 * 示例 2：
 * <p>
 * 输入：days = [1,2,3,4,5,6,7,8,9,10,30,31], costs = [2,7,15]
 * 输出：17
 * 解释：
 * 例如，这里有一种购买通行证的方法，可以让你完成你的旅行计划：
 * 在第 1 天，你花了 costs[2] = $15 买了一张为期 30 天的通行证，它将在第 1, 2, ..., 30 天生效。
 * 在第 31 天，你花了 costs[0] = $2 买了一张为期 1 天的通行证，它将在第 31 天生效。
 * 你总共花了 $17，并完成了你计划的每一天旅行。
 * <p>
 * <p>
 * 提示：
 * <p>
 * 1 <= days.length <= 365
 * 1 <= days[i] <= 365
 * days 按顺序严格递增
 * costs.length == 3
 * 1 <= costs[i] <= 1000
 */
public class LeetCode983_MincostTickets {

    int[] t = {1, 7, 30};

    public int mincostTickets(int[] days, int[] costs) {
        return f(days, costs, 0);
    }

    public int f(int[] days, int[] costs, int i) {
        if (i >= days.length) {
            return 0;
        }
        int ans = Integer.MAX_VALUE;
        for (int d = 0, j = i; d < t.length; d++) {
            // 选t后，下一个days的下标
            while (j < days.length && days[j] - days[i] < t[d]) {
                j++;
            }
            ans = Math.min(ans, costs[d] + f(days, costs, j));
        }
        return ans;
    }

    public int mincostTickets2(int[] days, int[] costs) {
        int[] dp = new int[days.length];
        Arrays.fill(dp, -1);
        return f2(days, costs, 0, dp);
    }

    // 记忆化搜索
    public int f2(int[] days, int[] costs, int i, int[] dp) {
        if (i >= days.length) {
            return 0;
        }

        if (dp[i] > 0) {
            return dp[i];
        }

        int ans = Integer.MAX_VALUE;
        for (int d = 0, j = i; d < t.length; d++) {
            // 选t后，下一个days的下标
            while (j < days.length && days[j] - days[i] < t[d]) {
                j++;
            }
            ans = Math.min(ans, costs[d] + f2(days, costs, j, dp));
        }

        dp[i] = ans;

        return ans;
    }


    public int mincostTickets3(int[] days, int[] costs) {
        return f3(days, costs);
    }

    public int f3(int[] days, int[] costs) {
        int[] dp = new int[366];
        Arrays.fill(dp, 0, days.length, Integer.MAX_VALUE);
        dp[days.length] = 0;

        for (int i = days.length - 1; i >= 0; i--) {
            for (int d = 0, j = i; d < t.length; d++) {
                while (j < days.length && days[j] - days[i] < t[d]) {
                    j++;
                }
                dp[i] = Math.min(dp[i], costs[d] + dp[j]);
            }
        }
        return dp[0];
    }

    public static void main(String[] args) {
        LeetCode983_MincostTickets leetCode983_mincostTickets = new LeetCode983_MincostTickets();
//        int i = leetCode983_mincostTickets.mincostTickets(new int[]{1, 4, 6, 7, 8, 20}, new int[]{2, 7, 15});
//        int i = leetCode983_mincostTickets.mincostTickets2(new int[]{1, 4, 6, 7, 8, 20}, new int[]{2, 7, 15});
        int i = leetCode983_mincostTickets.mincostTickets3(new int[]{1, 4, 6, 7, 8, 20}, new int[]{2, 7, 15});
        System.out.println("i = " + i);
    }
}
